package com.yang.repository;

import com.yang.entity.VideoTask;
import com.yang.type.DigitalHumanStatusEnum;
import com.yang.type.GenerateVideoActionEnum;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
import org.springframework.data.jpa.repository.Query;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

/**
 * 视频任务数据访问接口
 * 作者： yang
 * 时间： 2025/07/08
 */
@Repository
public interface VideoTaskRepository
        extends JpaRepository<VideoTask, Long>, JpaSpecificationExecutor<VideoTask> {
    /**
     * 根据任务状态、操作类型和失败次数查询视频任务列表，并按任务 ID 升序排序
     *
     * @param status 视频任务的状态，枚举类型，用于筛选符合指定状态的任务
     * @param action 视频任务的操作类型，枚举类型，用于筛选符合指定操作类型的任务
     * @param failedCount 视频任务的失败次数，筛选失败次数小于该值的任务
     * @return 符合条件的视频任务列表，按任务 ID 升序排列
     */
    List<VideoTask> findByStatusAndActionAndFailedCountLessThanOrderByIdAsc(
            DigitalHumanStatusEnum status, GenerateVideoActionEnum action, Integer failedCount);

    /**
     * 根据任务状态、操作类型和失败次数查找第一个符合条件的视频任务，并按任务 ID 升序排序
     *
     * @param status 视频任务的状态，枚举类型，用于筛选符合指定状态的任务
     * @param action 视频任务的操作类型，枚举类型，用于筛选符合指定操作类型的任务
     * @param failedCount 视频任务的失败次数，筛选失败次数小于该值的任务
     * @return 符合条件的第一个视频任务的 Optional 对象，如果未找到则返回空的 Optional
     */
    Optional<VideoTask> findFirstByStatusAndActionAndFailedCountLessThanOrderByIdAsc(
            DigitalHumanStatusEnum status, GenerateVideoActionEnum action, Integer failedCount);

    /**
     * 根据数字人 ID 查询视频任务
     *
     * @param taskId 视频任务的 ID，用于筛选符合指定任务 ID 的任务
     * @return 符合条件的视频任务列表
     */
    VideoTask findTaskByDigitalHumanId(Long taskId);
}
